import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;
import org.junit.jupiter.api.Test;

public class CodeGenerator {

    @Test
    public void run() {
        // 使用 FastAutoGenerator 快速配置代码生成器
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/education?serverTimezone=GMT%2B8", "root", "root")
                .globalConfig(builder -> {
                    builder.author("le") // 设置作者
                            .enableSwagger()
                            .disableOpenDir()
                            .dateType(DateType.ONLY_DATE)
                            .outputDir("D:\\workSpace\\educationM\\service\\ucenter_service" + "/src/main/java"); // 输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.leTest") // 设置父包名
                            .moduleName("ucenterService")
                            .entity("entity") // 设置实体类包名
                            .mapper("mapper") // 设置 Mapper 接口包名
                            .service("service") // 设置 Service 接口包名
                            .controller("controller"); // 设置 Mapper XML 文件包名
                })
                .strategyConfig(builder -> {
                    builder.addInclude("ucenter_member") // 设置需要生成的表名
                            .addTablePrefix("edu_")
                            .entityBuilder()
                            .naming(NamingStrategy.underline_to_camel)
                            .columnNaming(NamingStrategy.underline_to_camel)
                            .enableChainModel()
                            .idType(IdType.ASSIGN_ID)
                            .versionColumnName("version")
                            .logicDeleteColumnName("deleted")
                            .addTableFills(new Column("gmt_create", FieldFill.INSERT))
                            .addTableFills(new Column("gmt_modified", FieldFill.INSERT_UPDATE))
                            .formatFileName("%sEntity")
                            .enableLombok() // 启用 Lombok
                            .enableTableFieldAnnotation() // 启用字段注解
                            .controllerBuilder()
                            .enableHyphenStyle()
                            .enableRestStyle() // 启用 REST 风格
                            .serviceBuilder()
                            .formatServiceFileName("%sService");

                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用 Freemarker 模板引擎
                .execute(); // 执行生成
    }
}
